package com.yjzx.util.common.util.file;

import com.google.zxing.*;
import com.google.zxing.client.j2se.BufferedImageLuminanceSource;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import com.yjzx.util.common.util.ExceptionUtil;
import com.yjzx.util.common.util.StringUtils;

import javax.imageio.ImageIO;
import javax.swing.filechooser.FileSystemView;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

/**
 * @author yjzx
 * @date 2024/06/03
 * @description 二维码生成读取
 */
public class QrCodeUtil {

    /**
     * CODE_WIDTH：二维码宽度，单位像素
     */
    private static final int CODE_WIDTH = 300;
    /**
     * CODE_HEIGHT：二维码高度，单位像素
     */
    private static final int CODE_HEIGHT = 300;
    /**
     * 误差校正等级，L = ~7% correction、M = ~15% correction、Q = ~25% correction、H = ~30%
     */
    private static final ErrorCorrectionLevel ERROR_LEVEL = ErrorCorrectionLevel.M;
    /**
     * 嵌入的Logo的图片宽度(L(logo最大17%),M(logo最大22%)Q(logo最大25%)数据少会遮拦到定位，数据量大version越高logo可调大)
     */
    private static final int LOGO_WIDTH = CODE_WIDTH / 100 * 22;
    /**
     * 嵌入的Logo的图片高度(L(logo最大17%),M(logo最大22%)数据量大version越高可调大)
     */
    private static final int LOGO_HEIGHT = CODE_WIDTH / 100 * 22;
    /**
     * FRONT_COLOR：二维码前景色，0x000000 表示黑色
     */
    private static final int FRONT_COLOR = 0x000000;
    /**
     * BACKGROUND_COLOR：二维码背景色，0xFFFFFF 表示白色
     * 演示用 16 进制表示，和前端页面 CSS 的取色是一样的，注意前后景颜色应该对比明显，如常见的黑白
     */
    private static final int BACKGROUND_COLOR = 0xFFFFFF;

    /**
     * 将文本生成二维码，生成到指定路径
     *
     * @param content            文本
     * @param codeImgFileSaveDir 路径
     * @param fileName           文件名
     */
    public static void createCodeToFile(String content, File codeImgFileSaveDir, String fileName) {
        try {
            if (StringUtils.isBlank(content) || StringUtils.isBlank(fileName)) {
                return;
            }
            content = content.trim();
            if (codeImgFileSaveDir == null || codeImgFileSaveDir.isFile()) {
                //二维码图片存在目录为空，默认放在桌面...
                codeImgFileSaveDir = FileSystemView.getFileSystemView().getHomeDirectory();
            }
            if (!codeImgFileSaveDir.exists()) {
                //二维码图片存在目录不存在，开始创建...
                codeImgFileSaveDir.mkdirs();
            }

            //核心代码-生成二维码
            BufferedImage bufferedImage = getBufferedToLogoImage(content, null);

            File codeImgFile = new File(codeImgFileSaveDir, fileName);
            ImageIO.write(bufferedImage, "png", codeImgFile);
        } catch (Exception e) {
            throwException(e);
        }
    }

    /**
     * 生成二维码并输出到输出流
     *
     * @param text      ：二维码内容
     * @param outputStream ：输出流，比如 HttpServletResponse 的 getOutputStream
     */
    public static void createCodeToOutputStream(String text, OutputStream outputStream) {
        try {
            if (StringUtils.isBlank(text)) {
                return;
            }
            text = text.trim();
            //核心代码-生成二维码
            BufferedImage bufferedImage = getBufferedToLogoImage(text, null);

            //区别就是这一句，输出到输出流中，如果第三个参数是 File，则输出到文件中
            ImageIO.write(bufferedImage, "png", outputStream);
        } catch (Exception e) {
            throwException(e);
        }
    }

    /**
     * 根据文件内容，生成二维码
     * @param text
     * @return
     */
    public static BufferedImage getBufferedImage(String text){
        return getBufferedToLogoImage(text,null);
    }

    /**
     * 将文本和图片生成二维码
     *
     * @param text   文本内容
     * @param logoImage logo图片
     * @return
     * @throws WriterException 写入异常
     */
    public static BufferedImage getBufferedToLogoImage(String text, Image logoImage) {
        BufferedImage bufferedImage = null;
        try {
            //com.google.zxing.EncodeHintType：编码提示类型,枚举类型
            Map<EncodeHintType, Object> hints = new HashMap();

            //EncodeHintType.CHARACTER_SET：设置字符编码类型
            hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");

            //EncodeHintType.ERROR_CORRECTION：设置误差校正
            //ErrorCorrectionLevel：误差校正等级，L = ~7% correction、M = ~15% correction、Q = ~25% correction、H = ~30% correction
            //不设置时，默认为 L 等级，等级不一样，生成的图案不同，但扫描的结果是一样的
            hints.put(EncodeHintType.ERROR_CORRECTION, ERROR_LEVEL);

            //EncodeHintType.MARGIN：设置二维码边距，单位像素，值越小，二维码距离四周越近
            hints.put(EncodeHintType.MARGIN, 1);
            /**
             * MultiFormatWriter:多格式写入，这是一个工厂类，里面重载了两个 encode 方法，用于写入条形码或二维码
             *      encode(String contents,BarcodeFormat format,int width, int height,Map<EncodeHintType,?> hints)
             *      contents:条形码/二维码内容
             *      format：编码类型，如 条形码，二维码 等
             *      width：码的宽度
             *      height：码的高度
             *      hints：码内容的编码类型
             * BarcodeFormat：枚举该程序包已知的条形码格式，即创建何种码，如 1 维的条形码，2 维的二维码 等
             * BitMatrix：位(比特)矩阵或叫2D矩阵，也就是需要的二维码
             */
            MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
            BitMatrix bitMatrix = multiFormatWriter.encode(text, BarcodeFormat.QR_CODE, CODE_WIDTH, CODE_HEIGHT, hints);
            /**
             * java.awt.image.BufferedImage：具有图像数据的可访问缓冲图像，实现了 RenderedImage 接口
             * BitMatrix 的 get(int x, int y) 获取比特矩阵内容，指定位置有值，则返回true，将其设置为前景色，否则设置为背景色
             * BufferedImage 的 setRGB(int x, int y, int rgb) 方法设置图像像素
             *      x：像素位置的横坐标，即列
             *      y：像素位置的纵坐标，即行
             *      rgb：像素的值，采用 16 进制,如 0xFFFFFF 白色
             */
            bufferedImage = new BufferedImage(CODE_WIDTH, CODE_HEIGHT, BufferedImage.TYPE_INT_BGR);

            for (int x = 0; x < CODE_WIDTH; x++) {
                for (int y = 0; y < CODE_HEIGHT; y++) {
                    bufferedImage.setRGB(x, y, bitMatrix.get(x, y) ? FRONT_COLOR : BACKGROUND_COLOR);
                }
            }
            if (logoImage != null) {
                Graphics2D g = bufferedImage.createGraphics();
                g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
                g.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
                // 读取logo文件，计算正中央的位置，画到qrCode图片中去
                int logoX = (CODE_WIDTH - LOGO_WIDTH) / 2;
                int logoY = (CODE_HEIGHT - LOGO_HEIGHT) / 2;
                g.drawImage(logoImage, logoX, logoY, LOGO_WIDTH, LOGO_HEIGHT, null);

                g.dispose();
            }
        } catch (WriterException e) {
            throwException(e);
        }
        return bufferedImage;
    }

    /**
     * 根据本地二维码图片————————解析二维码内容
     * （注：图片必须是二维码图片，但也可以是微信用户二维码名片，上面有名称、头像也是可以的）
     *
     * @param file 本地二维码图片文件,如 E:\\logs\\2.jpg
     * @return
     */
    public static String parseQRCodeByFile(File file) {
        String resultStr = null;
        if (file == null || file.isDirectory() || !file.exists()) {
            return resultStr;
        }
        try {
            /**ImageIO 的 BufferedImage read(URL input) 方法用于读取网络图片文件转为内存缓冲图像
             * 同理还有：read(File input)、read(InputStream input)、、read(ImageInputStream stream)
             */
            BufferedImage bufferedImage = ImageIO.read(file);
            resultStr = parseQRCodeByImage(bufferedImage);
        } catch (IOException e) {
            throwException(e);
        }
        return resultStr;
    }

    /**
     * 根据网络二维码图片————————解析二维码内容
     * （区别仅仅在于 ImageIO.read(url); 这一个重载的方法）
     *
     * @param url 二维码图片网络地址，如 https://res.wx.qq.com/mpres/htmledition/images/mp_qrcode3a7b38.gif
     * @return
     */
    public static String parseQRCodeByUrl(URL url) {
        String resultStr = null;
        if (url == null) {
            return resultStr;
        }
        try {
            /**ImageIO 的 BufferedImage read(URL input) 方法用于读取网络图片文件转为内存缓冲图像
             * 同理还有：read(File input)、read(InputStream input)、、read(ImageInputStream stream)
             *
             * 如果图片网络地址错误，比如不能访问，则 read 抛异常：javax.imageio.IIOException: Can't get input stream from URL!
             */
            resultStr = parseQRCodeByImage(ImageIO.read(url));
        } catch (IOException e) {
            throwException(e);
        }


        return resultStr;
    }

    /**
     * 根据图片解析二维码内容
     *
     * @param image 图片
     * @return
     */
    public static String parseQRCodeByImage(BufferedImage image) {
        String resultStr = null;
        try {
            BufferedImageLuminanceSource source = new BufferedImageLuminanceSource(image);

            BinaryBitmap bitmap = new BinaryBitmap(new HybridBinarizer(source));
            HashMap<DecodeHintType, Object> hints = new HashMap<>();
            /**
             * 如果内容包含中文，则解码的字符集格式应该和编码时一致
             */
            hints.put(DecodeHintType.CHARACTER_SET, "UTF-8");
            /**
             * 如果图片不是二维码图片，则 decode 抛异常：com.google.zxing.NotFoundException
             * MultiFormatWriter 的 encode 用于对内容进行编码成 2D 矩阵
             * MultiFormatReader 的 decode 用于读取二进制位图数据
             */
            Result result = new MultiFormatReader().decode(bitmap, hints);
            resultStr = result.getText();
        } catch (NotFoundException e) {
            throwException(e);
        }
        return resultStr;
    }


    public static void throwException(Exception exception) {
        ExceptionUtil.throwToRunException(exception,RuntimeException.class);
    }


}
